import pandas as pd
import numpy as np
import os
import datetime
codes=['000001.SZ',
'000002.SZ',
'000728.SZ',
'002007.SZ',
'002032.SZ',
'002142.SZ',
'002230.SZ',
'002271.SZ',
'002352.SZ',
'002456.SZ',
'002555.SZ',
'002714.SZ',
'300003.SZ',
'300015.SZ',
'300017.SZ',
'300142.SZ',
'600000.SH',
'600048.SH',
'600066.SH',
'600111.SH',
'600176.SH',
'600346.SH',
'600369.SH',
'600383.SH',
'600487.SH',
'600516.SH',
'600547.SH',
'600585.SH',
'600663.SH',
'600760.SH',
'600837.SH',
'601012.SH',
'601138.SH',
'601377.SH',
'601555.SH',
'601577.SH',
'601688.SH',
'601727.SH',
'601808.SH',
'601838.SH',
'601888.SH',
'601933.SH',
'603986.SH']
#modify=pd.DataFrame(columns=['基金名称','投资组合名称','市场','证券代码','股东代码','托管席位','投资类型','目标组合名称','目标投资类型','业务类型','发生数量','成本','价格','操作生效日期','操作失效日期','备注','调整类型','变动成本','变动收益','多空标志','是否影响标准券','组合编号','目标组合编号','成本调整类型','人民币变动成本','收益调整类型','人民币变动收益','是否受限股调整','受限类型','锁定受限类型','受限结束日','基金代码'])
accountInfo1={'基金名称':'11003','投资组合名称':'缺省组合','市场':'上交所A','股东代码':'D890019178','托管席位':'33211'}
accountInfo2={'基金名称':'11003','投资组合名称':'缺省组合','市场':'深交所A','股东代码':'0899017364','托管席位':'061700'}
accountInfo3={'基金名称':'11038','投资组合名称':'38组合','市场':'上交所A','股东代码':'D890192510','托管席位':'33211'}
accountInfo4={'基金名称':'11038','投资组合名称':'38组合','市场':'深交所A','股东代码':'0899017686','托管席位':'061700'}
today=datetime.date.today()
todayStr=today.strftime("%Y%m%d")
num=0
xlsname=f"Stock_{todayStr}_{num}.xls"
sheetname=f"Stock_{todayStr}_{num}"
mylist=[]
for code in codes:
    codeStr=code.split('.')[0]
    market=code.split('.')[1]
    dict={}
    if market=='SH':
        dict=accountInfo1
        dict.update({'证券代码':codeStr,'业务类型':'证券增加','发生数量':100000,'操作生效日期':todayStr,'备注':'11003账户缺省组合增加证券'})
        pass
    if market=='SZ':
        dict=accountInfo2
        dict.update({'证券代码':codeStr,'业务类型':'证券增加','发生数量':100000,'操作生效日期':todayStr,'备注':'11003账户缺省组合增加证券'})
        pass
    mylist.append(dict)
    pass
modify=pd.DataFrame(mylist)
modify.to_excel(xlsname,index_label=None)
